#!/usr/bin/env python
# encoding: utf-8
"""
@author: Alan
@file: train.py
@time: 2020/7/28 13:04
@desc:
"""

from tensorflow.keras.models import Model
from tensorflow.keras.layers import Input
from tensorflow.keras.optimizers import Adam
from yolov3_by_myself.yolov3_body import body
from yolov3_by_myself.get_yture import generator

num_anchors = 3
num_classes = 5

model_input = Input(shape=(416, 416, 3))
model_output = body(model_input, num_anchors, num_classes)
model = Model(model_input, model_output)

model.compile(optimizer=Adam(lr = 1e-3),loss=)

model.fit_generator(generator())
